Sorting numpy array according to index provided by another array